from langchain_core.prompts import ChatPromptTemplate
from langchain_ollama import ChatOllama

from common_config import CHAT_OLLAMA_MODEL

llm = ChatOllama(model=CHAT_OLLAMA_MODEL, temperature=0.1)

template = """
新信息：2025年美国现任（第47任）总统是特朗普
美国现任（2025）的总统是谁: 特朗普

{message}

/nothink
"""

prompt = ChatPromptTemplate.from_template(template)

chat = prompt | llm

for chuck in chat.stream({"message": """"
                                    回答下面两个问题：
                                    1. 美国的总统是谁？
                                    2. 特朗普是谁？"""}):
    print(chuck.content, end="", flush=True)

